Skip to content

Prepare for JAX 0.9.1 release#338

Open
gulsumgudukbay wants to merge 14 commits intorocm-jaxlib-v0.9.1from
release-0.9.1-prep
Open

Prepare for JAX 0.9.1 release#338
gulsumgudukbay wants to merge 14 commits intorocm-jaxlib-v0.9.1from
release-0.9.1-prep

Conversation

@gulsumgudukbay
Copy link
Contributor

Motivation

Preparing the repo for 0.9.1 release

Technical Details

In progress...

Test Plan

In progress...

Test Result

In progress...

Submission Checklist

gulsumgudukbay and others added 14 commits March 2, 2026 20:33
…-encoding fix (#335)

* Cherry-pick PR #293 + #306: dev tarball Docker tag fix and double URL-encoding fix

- build/ci_build: replace '+' with '.' in rocm_ver_tag and rocm_version_tag for valid Docker tags
- jax_rocm_plugin/build/rocm/ci_build: ROCM_VERSION_EXTRA use replace('+', '.')
- tools/get_rocm.py: unquote then quote therock URL to avoid double-encoding; create amdgcn symlink only if missing

Made-with: Cursor

* Fix pylint C0301: shorten comment line (line-too-long 103/100)

Made-with: Cursor

* Use try/except FileExistsError for amdgcn symlink (PR #330)

Made-with: Cursor

---------

Co-authored-by: Jenkins <jenkins-compute@amd.com>
(cherry picked from commit ea9a021)
On WSL, ROCm GPU discovery via KFD topology is unavailable, which can
cause plugin initialization to fail.
This change adds a WSL-specific fast path in GPU counting: when /dev/dxg
exists, it returns 1 directly for initialization gating.
The existing KFD-based counting logic remains unchanged for other
environments, and the docstring is updated to clearly document this
intentional hardcoded behavior.

Signed-off-by: Flora Cui <flora.cui@amd.com>
(cherry picked from commit ab4424b)
(cherry picked from commit 33fbb4b)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

6 participants